import pandas as pd
import numpy as np

data = {'A': [1, 2, 3, 4], 'B': [10, 20, 30, 40], 'C': [100, 200, 300, 400]}
df = pd.DataFrame(data, index=['row1', 'row2', 'row3', 'row4'])
print(df)

# 可以通过列名直接选择单列或多列的数据。
# 语法
# df['column_name']    # 单列选择
# df[['col1', 'col2']]  # 多列选择

# 单列选择：通过 df['A']选择单列数据，返回一个 `Series`。
# 多列选择：通过 df[['A', 'B']] 选择多列数据，返回一个 `DataFrame`

# 选择 A 列
print(type(df['A']))
print(df['A'])

# 选择 A 列和 B 列
print(df[['A', 'B']])